package main

import (
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"github.com/jmoiron/sqlx" // 新加依赖包
)

// sqlx实现批量操作数据

var db *sqlx.DB // 一个连接池对象

func initDB() (err error) {
	dsn := "root:123456@tcp(10.10.10.101:33306)/test_demo"
	db, err = sqlx.Connect("mysql", dsn)
	if err != nil { // dsn格式不正确
		fmt.Printf("dsn: %s invalid, err: %v\n", dsn, err)
		return
	}
	err = db.Ping()
	if err != nil { // 连接不上数据库,说明用户名密码错误
		fmt.Printf("open %s failed, err: %v\n", dsn, err)
		return
	}
	// 设置数据库连接池最大连接数
	db.SetMaxOpenConns(10)
	// 最大闲置连接数
	db.SetMaxIdleConns(5)
	return
}

type user struct {
	Id   int
	Name string
	Age  int
}

func main() {
	err := initDB()
	if err != nil {
		fmt.Printf("init DB failed, err: %v\n", err)
		return
	}
	fmt.Println("连接数据库成功")

	sqlStr1 := `select id,name,age from user where id = ?`
	var u user
	err = db.Get(&u, sqlStr1, 1)
	if err != nil {
		fmt.Printf("get failed, err: %v\n", err)
		return
	}
	fmt.Printf("u:%#v\n", u)

	// 查询所有数据
	var userList []user
	sqlStr2 := `select id,name,age from user`
	db.Select(&userList, sqlStr2)
	for _, v := range userList {
		fmt.Println(v)
	}
}
